use crate::Var;

#[derive(Debug, Clone, Copy)]
pub enum Activation {
    Sigmoid,
    ReLU,
    Tanh,
}

impl Activation {
    pub fn forward(&self, var: &Var) -> Var {
        match self {
            Self::Sigmoid => var.sigmoid(),
            Self::ReLU => var.relu(),
            Self::Tanh => var.tanh(),
        }
    } 
}